package com.jiangnanxiaoguan.controller;


import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.jiangnanxiaoguan.common.Result;
import com.jiangnanxiaoguan.dto.SetmealDto;
import com.jiangnanxiaoguan.entity.Category;
import com.jiangnanxiaoguan.entity.Setmeal;
import com.jiangnanxiaoguan.entity.SetmealDish;
import com.jiangnanxiaoguan.service.CategoryService;
import com.jiangnanxiaoguan.service.SetmealDishService;
import com.jiangnanxiaoguan.service.SetmealService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cache.annotation.CacheEvict;
import org.springframework.cache.annotation.Cacheable;
import org.springframework.web.bind.annotation.*;

import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * <p>
 * 套餐 前端控制器
 * </p>
 *
 * @author 不愿透露姓名的狂热者
 * @since 2022-09-11
 */
@Slf4j
@RestController
@RequestMapping("/setmeal")
public class SetmealController {
    @Autowired
    private SetmealService setmealService;

    @Autowired
    private CategoryService categoryService;

    @Autowired
    private SetmealDishService setmealDishService;

    /**
     * 套餐分页查询。
     *
     * @param page
     * @param pageSize
     * @param name
     * @return
     */
    @GetMapping("/page")
    public Result<IPage<Setmeal>> page(Integer page, Integer pageSize, String name) {
        IPage<Setmeal> setmealPage = new Page<>(page, pageSize);
        LambdaQueryWrapper<Setmeal> setmealLambdaQueryWrapper = new LambdaQueryWrapper<>();
        setmealLambdaQueryWrapper.like(StringUtils.isNotEmpty(name), Setmeal::getName, name);
        setmealLambdaQueryWrapper.orderByDesc(Setmeal::getUpdateTime);
        setmealService.page(setmealPage, setmealLambdaQueryWrapper);

        // 这种方式比较少见（效率较慢，连接数据库次数较多），可使用实体类增加冗余字段的方式，那是实际开发会使用的。
//        Page<SetmealDto> setmealDtoPage = new Page<>();
//        BeanUtils.copyProperties(setmealPage, setmealDtoPage, "records");
//
//        List<Setmeal> setmeals = setmealPage.getRecords();
//        Stream<SetmealDto> setmealDtoStream = setmeals.stream().map(setmeal -> {
//            SetmealDto setmealDto = new SetmealDto();
//            BeanUtils.copyProperties(setmeal, setmealDto);
//
//            Category category = categoryService.getById(setmeal.getCategoryId());
//
//            if (category != null) {
//                setmealDto.setCategoryName(category.getName());
//            }
//            return setmealDto;
//        });
//
//        setmealDtoPage.setRecords(setmealDtoStream.collect(Collectors.toList()));


        // 开发中使用的方式：增加冗余字段，最后给其赋值。（直接从DishController复制粘贴过来的，改一下就完事）
        // 取出所有菜品
        List<Setmeal> setmeals = setmealPage.getRecords();
        // stream得到这些菜品的id
        Set<Long> categoryIds = setmeals.stream().map(Setmeal::getCategoryId).collect(Collectors.toSet());
        // 批量查询，得到所有菜品的分类信息
        List<Category> categories = categoryService.listByIds(categoryIds);
        // 得到分类id对应分类名的map集合
        Map<Long, String> categoryNamesMap = categories.stream().collect(Collectors.toMap(Category::getId, Category::getName));
        // 给每个dish的分类名赋值
        for (Setmeal setmeal : setmeals) {
            setmeal.setCategoryName(categoryNamesMap.get(setmeal.getCategoryId()));
        }
        // 更新分页信息
        setmealPage.setRecords(setmeals);

        return Result.success(setmealPage);
    }

    /**
     * 添加套餐。
     *
     * @param setmealDto
     * @return
     */
    @CacheEvict(cacheNames = "setmealCache", allEntries = true)
    @PostMapping
    public Result<String> save(@RequestBody SetmealDto setmealDto) {
        log.info(setmealDto.toString());
        setmealService.saveWithSetmealDishes(setmealDto);
        return Result.success("保存套餐成功");
    }

    /**
     * 删除/批量删除套餐。
     *
     * @param ids
     * @return
     */
    @CacheEvict(cacheNames = "setmealCache", allEntries = true)
    @DeleteMapping
    public Result<String> delete(@RequestParam List<Long> ids) {
        log.info(ids.toString());
        setmealService.deleteWithSetmealDishes(ids);
        return Result.success("删除套餐成功");
    }

    /**
     * 用户端套餐查询显示。
     *
     * @param setmeal
     * @return
     */
    @Cacheable(cacheNames = "setmealCache")
    @GetMapping("/list")
    public Result<List<Setmeal>> list(Setmeal setmeal) {
        log.info("套餐信息：" + setmeal.toString());
        Long categoryId = setmeal.getCategoryId();

        LambdaQueryWrapper<Setmeal> setmealLambdaQueryWrapper = new LambdaQueryWrapper<>();
        setmealLambdaQueryWrapper.eq(categoryId != null, Setmeal::getCategoryId, categoryId);
        setmealLambdaQueryWrapper.eq(Setmeal::getStatus, 1);

        List<Setmeal> setmealList = setmealService.list(setmealLambdaQueryWrapper);

        return Result.success(setmealList);
    }

    /**
     * （批量）停售/起售套餐。
     *
     * @param statusNumber
     * @param ids
     * @return
     */
    @CacheEvict(cacheNames = "setmealCache", allEntries = true)
    @PostMapping("/status/{statusNumber}")
    public Result<String> updateStatus(@PathVariable Integer statusNumber, @RequestParam List<Long> ids) {
        log.info("套餐状态：{}, 套餐id：{}", statusNumber, ids);

        // 先查询出所有需要修改的套餐
        List<Setmeal> setmealList = setmealService.listByIds(ids);

        // TODO: 优化，将套餐状态值已经是将要修改的状态值的套餐，从集合中剔除。
        List<Setmeal> setmeals = setmealList.stream().
                filter(setmeal -> !statusNumber.equals(setmeal.getStatus())).collect(Collectors.toList());

//        setmealList.stream().forEach(setmeal -> setmeal.setStatus(statusNumber));
        // 遍历套餐，将状态设置为需要修改的装填
        setmeals.forEach(setmeal -> {
            // TODO: 非优化，更新时依然是这个集合，SQL语句没有减少。
//            if (!statusNumber.equals(setmeal.getStatus())) {
            setmeal.setStatus(statusNumber);
//            }
        });

        // 批量修改数据库中套餐表状态（TODO: 只更新需要更新状态的套餐，否则前面的优化是空谈。）
        setmealService.updateBatchById(setmeals);

        return Result.success("");
    }

    /**
     * 根据套餐id获取套餐，以及套餐菜品，用于修改套餐回显信息。
     *
     * @param id
     * @return
     */
    @GetMapping("/{id}")
    public Result<SetmealDto> getById(@PathVariable Long id) {
        log.info("[套餐回显] 套餐id：{}", id);

        // 1、先查询出套餐基本信息
        Setmeal setmeal = setmealService.getById(id);

        // 2、将套餐基本信息拷贝到dto
        SetmealDto setmealDto = new SetmealDto();
        BeanUtils.copyProperties(setmeal, setmealDto);

        // 3、通过套餐id，从套餐菜品关系表中查询出该套餐中所有菜品
        LambdaQueryWrapper<SetmealDish> setmealDishLambdaQueryWrapper = new LambdaQueryWrapper<>();
        setmealDishLambdaQueryWrapper.eq(SetmealDish::getSetmealId, id);
        List<SetmealDish> setmealDishes = setmealDishService.list(setmealDishLambdaQueryWrapper);

        // 4、将套餐菜品追加到dto
        setmealDto.setSetmealDishes(setmealDishes);

        return Result.success(setmealDto);
    }

    /**
     * 修改套餐 / 修改或添加套餐菜品。
     *
     * @param setmealDto
     * @return
     */
    @CacheEvict(cacheNames = "setmealCache", allEntries = true)
    @PutMapping
    public Result<String> update(@RequestBody SetmealDto setmealDto) {
        log.info("[套餐修改] 套餐所有菜品信息：{}", setmealDto);

        setmealService.updateWithSetmealDishes(setmealDto);

        return Result.success("");
    }
}

